# server_calculator.py
from mcp.server.fastmcp import FastMCP
from typing import List

from mcp_test.two_agent import add_two

mcp = FastMCP(name="CalculatorServer")
mcp.add_tool(add_two, name="AddTwo",description="Add two numbers")

@mcp.tool()
def add(a: int, b: int) -> int:
    """计算两个整数的和"""
    return a + b


@mcp.tool()
def get_stus() -> List[str]:
    """获取学生列表"""
    sts = []
    sts.append("张三")
    sts.append("李四")
    sts.append("王五")
    return sts

@mcp.resource("greeting://{name}")
def get_greeting(name: str) -> str:
    """获取个性化问候语"""
    return f"Hello, {name}! 欢迎使用 MCP 服务。"

if __name__ == "__main__":
    mcp.run(transport="sse")  # 不同工具可使用不同端口